import os
from PIL import Image

# 设置路径A, C, D
path_a = r'D:\Desktop\RoboMaster-Unity\Screenshots'  # 输入图片文件夹路径A
path_c = r'D:\Desktop\RoboMaster-Unity\Screenshots\left'  # 输出左半部分图片的文件夹路径C
path_d = r'D:\Desktop\RoboMaster-Unity\Screenshots\right'  # 输出右半部分图片的文件夹路径D

# 如果输出文件夹不存在，创建文件夹
os.makedirs(path_c, exist_ok=True)
os.makedirs(path_d, exist_ok=True)

# 遍历路径A中的所有文件
for filename in os.listdir(path_a):
    file_path = os.path.join(path_a, filename)

    # 仅处理图片文件
    if os.path.isfile(file_path) and filename.lower().endswith(('jpg', 'jpeg', 'png', 'bmp', 'tiff')):
        with Image.open(file_path) as img:
            # 获取图片的宽度和高度
            width, height = img.size

            # 分割图片为左右两部分
            left = img.crop((0, 0, width // 2, height))  # 左半部分
            right = img.crop((width // 2, 0, width, height))  # 右半部分

            # 保存左半部分到路径C，右半部分到路径D
            left.save(os.path.join(path_c, filename))
            right.save(os.path.join(path_d, filename))

        print(f'Processed: {filename}')
